#!/usr/bin/env python3

import os
import re
import argparse

if __name__ == '__main__':
    
    parser = argparse.ArgumentParser(description='使用正则表达式替换文件名称')
    parser.add_argument('--re', type=str, help='正则表达式')
    parser.add_argument('--out', type=str, help='输出文件名称 (不包含路径) 可以使用\1 \2等表示匹配的组')
    args = parser.parse_args()
    root = os.getcwd()
    print(root)
    pattern = re.compile(args.re)
    for file in os.listdir(root):
        if re.match(args.re,file) and os.path.isfile(file): 
            newfile = re.sub(pattern,args.out,file)
            print(os.path.join(root,file) ,'->',os.path.join(root,newfile)) 
            os.rename(os.path.join(root,file),os.path.join(root,newfile))
